from sqlalchemy import Column, ForeignKey, String, Integer, Text, orm
from app.models.base import Base

class Experiment(Base):
    E_id = Column(Integer, primary_key=True,autoincrement=True)

    T_id = Column(Integer, nullable=False)
    course_number = Column(String(50), nullable=False)
    question_text = Column(Text, nullable=True)
    question_image_path = Column(String(255), nullable=True)
    answer_image_path = Column(String(255), nullable=True)
    LLMname = Column(String(50), nullable=False)
    score = Column(Integer, nullable=False)
    comments = Column(Text, nullable=False)
    def __init__(self,T_id,course_number,question_text,question_image_path,answer_image_path,LLMname,score,comments):
        super(Experiment,self).__init__()
        self.T_id = T_id
        self.course_number = course_number
        self.question_text = question_text
        self.question_image_path = question_image_path
        self.answer_image_path = answer_image_path
        self.LLMname = LLMname
        self.score = score
        self.comments = comments